package com.dyna.service.impl;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.core.util.ObjUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.extra.spring.SpringUtil;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.dyna.constants.ModuleErrorCodeConstants;
import com.dyna.core.exception.CustomException;
import com.dyna.domain.resp.PageResult;
import com.dyna.enums.CommonStatusEnum;
import com.dyna.enums.DataScopeEnum;
import com.dyna.enums.RoleCodeEnum;
import com.dyna.enums.RoleTypeEnum;
import com.dyna.foundation.domain.entity.SysRole;
import com.dyna.foundation.domain.form.SysRoleSaveForm;
import com.dyna.foundation.domain.req.SysRolePageReq;
import com.dyna.mapper.SysRoleMapper;
import com.dyna.service.SysPermissionService;
import com.dyna.service.SysRoleService;
import com.google.common.annotations.VisibleForTesting;
import lombok.AllArgsConstructor;
import org.springframework.context.annotation.Lazy;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.StringUtils;

import javax.annotation.Resource;
import java.util.Collection;
import java.util.List;

/**
* @author zhoucaiwang
*/
@Service
@AllArgsConstructor
public class SysRoleServiceImpl extends ServiceImpl<SysRoleMapper, SysRole> implements SysRoleService {

    @Lazy
    @Resource
    private SysPermissionService permissionService;

    @Override
    @Transactional(rollbackFor = Exception.class)
    public Long createRole(SysRoleSaveForm createReqVO, Integer type) {
        // 1. 校验角色
        validateRoleDuplicate(createReqVO.getName(), createReqVO.getCode(), null);

        // 2. 插入到数据库
        SysRole role = BeanUtil.toBean(createReqVO, SysRole.class)
                .setType(ObjectUtil.defaultIfNull(type, RoleTypeEnum.CUSTOM.getType()))
                .setStatus(ObjUtil.defaultIfNull(createReqVO.getStatus(), CommonStatusEnum.ENABLE.getStatus()))
                .setDataScope(DataScopeEnum.ALL.getScope()); // 默认可查看所有数据。原因是，可能一些项目不需要项目权限
        baseMapper.insert(role);
        return role.getId();
    }

    @Override
    public void updateRole(SysRoleSaveForm updateReqVO) {
        // 1.1 校验是否可以更新
        SysRole role = validateRoleForUpdate(updateReqVO.getId());
        // 1.2 校验角色的唯一字段是否重复
        validateRoleDuplicate(updateReqVO.getName(), updateReqVO.getCode(), updateReqVO.getId());

        // 2. 更新到数据库
        SysRole updateObj = BeanUtil.toBean(updateReqVO, SysRole.class);
        baseMapper.updateById(updateObj);
    }

    /**
     * 校验角色是否可以被更新
     *
     * @param id 角色编号
     */
    @VisibleForTesting
    SysRole validateRoleForUpdate(Long id) {
        SysRole role = baseMapper.selectById(id);
        if (role == null) {
            throw new CustomException(ModuleErrorCodeConstants.ROLE_NOT_EXISTS.getMsg());
        }
        // 内置角色，不允许删除
        if (RoleTypeEnum.SYSTEM.getType().equals(role.getType())) {
            throw new CustomException(ModuleErrorCodeConstants.ROLE_CAN_NOT_UPDATE_SYSTEM_TYPE_ROLE.getMsg());
        }
        return role;
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void deleteRole(Long id) {
        // 1. 校验是否可以更新
        SysRole role = validateRoleForUpdate(id);

        // 2.1 标记删除
        baseMapper.deleteById(id);
        // 2.2 删除相关数据
        permissionService.processRoleDeleted(id);
    }

    @Override
    public SysRole getRole(Long id) {
        return baseMapper.selectById(id);
    }

    @Override
    public PageResult<SysRole> getRolePage(SysRolePageReq reqVO) {
        return baseMapper.selectPage(reqVO);
    }

    @Override
    public List<SysRole> getRoleListByStatus(Collection<Integer> statuses) {
        return baseMapper.selectListByStatus(statuses);
    }

    /**
     * 校验角色的唯一字段是否重复
     *
     * 1. 是否存在相同名字的角色
     * 2. 是否存在相同编码的角色
     *
     * @param name 角色名字
     * @param code 角色额编码
     * @param id 角色编号
     */
    @VisibleForTesting
    void validateRoleDuplicate(String name, String code, Long id) {
        // 0. 超级管理员，不允许创建
        if (RoleCodeEnum.isSuperAdmin(code)) {
            throw new CustomException(StrUtil.format(ModuleErrorCodeConstants.ROLE_ADMIN_CODE_ERROR.getMsg(), code));
        }
        // 1. 该 name 名字被其它角色所使用
        SysRole role = baseMapper.selectByName(name);
        if (role != null && !role.getId().equals(id)) {
            throw new CustomException(StrUtil.format(ModuleErrorCodeConstants.ROLE_NAME_DUPLICATE.getMsg(), name));
        }
        // 2. 是否存在相同编码的角色
        if (!StringUtils.hasText(code)) {
            return;
        }
        // 该 code 编码被其它角色所使用
        role = baseMapper.selectByCode(code);
        if (role != null && !role.getId().equals(id)) {
            throw new CustomException(StrUtil.format(ModuleErrorCodeConstants.ROLE_CODE_DUPLICATE.getMsg(), code));
        }
    }

    @Override
    public List<SysRole> getRoleList() {
        return baseMapper.selectList();
    }

    @Override
    public SysRole getRoleFromCache(Long id) {
        return baseMapper.selectById(id);
    }

    @Override
    public boolean hasAnySuperAdmin(Collection<Long> ids) {
        if (CollectionUtil.isEmpty(ids)) {
            return false;
        }
        SysRoleServiceImpl self = getSelf();
        return ids.stream().anyMatch(id -> {
            SysRole role = self.getRoleFromCache(id);
            return role != null && RoleCodeEnum.isSuperAdmin(role.getCode());
        });
    }

    /**
     * 获得自身的代理对象，解决 AOP 生效问题
     *
     * @return 自己
     */
    private SysRoleServiceImpl getSelf() {
        return SpringUtil.getBean(getClass());
    }

}




